// 封装指令
import Vue from 'vue'

export default {
  install () {
    Vue.directive('fofo', {
      // 页面插入DOM时自动聚焦
      inserted (el) {
        fn(el)
      },
      // 页面更新DOM时自动聚焦
      update (el) {
        fn(el)
      }
    })
  }
}
function fn (el) {
  // 这里必须大写
  if (el.nodeName === 'TEXTAREA' || el.nodeName === 'INPUT') {
    el.focus()
  } else {
    const theInput = el.querySelector('input')
    const theArea = el.querySelector('textarea')
    // 需要判断是否为空才能聚焦 否则会报错
    if (theInput) theInput.focus()
    if (theArea) theArea.focus()
  }
}
